package com.xj.mq.mqserver.datacenter;

import com.xj.mq.MqApplication;
import com.xj.mq.mqserver.core.Binding;
import com.xj.mq.mqserver.core.Exchange;
import com.xj.mq.mqserver.core.ExchangeType;
import com.xj.mq.mqserver.core.MessageQueue;
import com.xj.mq.mqserver.mapper.MetaMapper;

import java.io.File;
import java.util.List;
import java.util.Queue;

/**
 * Created with IntelliJ IDEA
 * Description 整合数据库操作
 *              进一步封装 如果不用MyBatis 不用改变此类的接口 只需改变内部的实现
 * User: 王杰
 * Date: 2025-05-14
 * Time: 23:49
 */
public class DataBaseManager {
    // 要从 Spring 中拿到现成的对象
    private MetaMapper metaMapper;

    public void init() {
        // 手动的获取到 MetaMapper
        metaMapper = MqApplication.context.getBean(MetaMapper.class);

        if (!checkDBExists()) {
            // 数据库不存在 建库建表
            // 先创建一个 data 目录
            File file = new File("./data");
            file.mkdirs();
            // 创建数据表
            createTable();
            // 插入默认数据
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成");
        }else {
            // 数据库已经存在 什么都不做
            System.out.println("[DataBaseManager] 数据库已存在");
        }
    }

    // 删除数据库
    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库文件成功！");
        } else {
            System.out.println("[DataBaseManager] 删除数据库文件失败！");
        }

        File dataDir = new File("./data");
        // 使用 delete 删除目录的时候 目录必须是空的
        ret = dataDir.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库目录成功！");
        } else {
            System.out.println("[DataBaseManager] 删除数据库目录失败！");
        }
    }



    // 判断数据库是否存在
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        if (file.exists()) {
            return true;
        }
        return false;
    }

    // 建表
    // 建库操作并不需要手动执行（不需要手动创建meta.db文件,MyBatis帮我们完成了）
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成!");
    }

    // 添加默认数据
    // 主要：添加一个默认的交换机
    // RabbitMQ 里的设定：带有一个匿名的交换机, 类型是 DIRECT
    private void createDefaultData() {
        // 构造一个默认的交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setExchangeType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 创建初始数据完成!");
    }

    // 把其他的数据库操作, 也在这个类中封装一下
    public void insertExchange(Exchange exchange) {
        metaMapper.insertExchange(exchange);
    }

    public List<Exchange> selectAllExchanges() {
        return metaMapper.selectAllExchanges();
    }

    public void deleteExchange(String exchangeName) {
        metaMapper.deleteExchange(exchangeName);
    }

    public void insertQueue(MessageQueue messageQueue) {
        metaMapper.insertQueue(messageQueue);
    }

    public List<MessageQueue> selectAllQueues() {
        return metaMapper.selectAllQueues();
    }

    public void deleteQueue(String messageQueueName) {
        metaMapper.deleteQueue(messageQueueName);
    }

    public void insertBinding(Binding binding) {
        metaMapper.insertBinding(binding);
    }

    public List<Binding> selectAllBindings() {
        return metaMapper.selectAllBindings();
    }

    public void deleteBinding(Binding binding) {
        metaMapper.deleteBinding(binding);
    }


}
